from db.model import Model#导入定义自定义类

if __name__=='__main__':#判断当前程序是否为主程序
    mod=Model('atm')#将数据表的名字传入类
    while True:#主循环
        account=input('请输入账号: ')#输入账号名
        if mod.find(account)==None:#判断该账号是否存在
            print('此账号不存在')

        password=int(input('请输入密码： '))#输入密码
        if password==mod.find(account)['password']:#判断密码是否相符
            while True:#定义用户操作循环
                print('1.查询余额 2.取钱 3.存钱 4.退出')
                key=input('请输入')#输入用户操作
                while key not in ['1','2','3','4']:#判断是否为无效输入
                    print('无效的输入 请重新输入')
                    key=input('请输入')
                if key=='1':
                    print('您的账户余额为： ',mod.find(account)['money'])#查询返回字典中的钱的数值
                if key=='2':
                    withdraw=int(input('请输入取钱金额： '))
                    while withdraw>mod.find(account)['money']:#判断取钱金额是否大于账户余额
                        print('您的余额不足')
                        withdraw = int(input('请重新输入取钱金额： '))
                    after=mod.find(account)['money']-withdraw#将账户余额减去取钱金额
                    mod.update({'money':after,'id':account})#更新金额数值
                    print('您的账户余额为： ',mod.find(account)['money'])

                if key=='3':
                    savemoney=int(input('请输入存钱金额： '))
                    while savemoney<=0:#判断输入数值是否有效
                        print('无效的金额')
                        savemoney = input('请重新输入存钱金额： ')
                    aftersave=mod.find(account)['money']+savemoney#将金额相加
                    mod.update({'money':aftersave,'id':account})#更新金额数值
                    print('您的账户余额为： ',aftersave)
                if key=='4':
                    print('欢迎下次使用！')
                    break#退出循环
        else:
            print('密码错误')